import streamlit as st
import re
import data.Data as dd
import time
#设置标签页
st.set_page_config(
    page_title="Register",
    page_icon="👏"
)
#设置标题
st.title("私人助手register 😊😊")

#设置注册页面的组件
username = st.text_input('请输入你的账号')
password = st.text_input('请输入你的密码',type='password')
repass = st.text_input('请再次输入你的密码',type='password')
registerFlag = st.button('注册')

loginFlag = st.button("已有帐号，点击登录")

#def query_user(username):
#定义一个注册函数
def register(username, password,repass):
    #1.校验信息是否填写
    if username and password and repass:
        #2.校验用户名长度是否为11位   且是否为手机号  正则表达式
        if re.match('^(13|15|17|18|19)[0-9]{9}$', username):
            #3.手机号长度8位且输入密码一致
            if password == repass and len(password) >= 8:
                #4.查询数据库是否有重复信息
                if dd.query_user_no_exits(username):
                    dd.add_user(username,password)
                    st.success("注册成功")
                    time.sleep(2)
                    st.switch_page("login.py")
                else:
                    st.error("用户已注册，请勿重复注册")
            else:
                st.error("两次密码不一致或者密码长度不够8位")
        else:
            st.error("手机号格式不正确")

    else:
        st.error("请务必填写相关注册信息")


#拿到密码，校验信息填写，校验账号是否存在，校验两次密码是否一致
#都成功，添加数据库再跳转登录页面
if registerFlag:
    register(username,password,repass)

#当点击了登录按钮，跳转登录界面
if loginFlag:
    #跳转到系统首页，不加pages
    st.switch_page("login.py")
